class Config():
    def __init__(self):
        self.root_path = "D:/黑马/PediatricDepartment/"
        # 原始数据路径
        self.train_datapath = self.root_path + 'data/pediatric_train.csv'
        self.test_datapath = self.root_path + 'data/pediatric_test.csv'
        self.dev_datapath = self.root_path + 'data/pediatric_dev.csv'
        # 类别文档
        self.class_doc_path = self.root_path + "data/class.txt"
        # 停用词路径
        self.stop_words_path = self.root_path + "data/stopwords.txt"

        # 处理后的数据路径
        self.process_train_datapath = self.root_path + "random_forest/final_data/train_process.csv"
        self.process_test_datapath = self.root_path + "random_forest/final_data/test_process.csv"
        self.process_dev_datapath = self.root_path + "random_forest/final_data/dev_process.csv"

        # 保存模型路径
        self.random_forest_model_save_path = self.root_path + r"random_forest/save_models/random_forest_model.pkl"
        self.tfidf_model_save_path = self.root_path + r"random_forest/save_models/tfidf_model.pkl"
        # 模型预测结果
        self.model_predict_result = self.root_path + r"random_forest/result/predict_result.txt"


if __name__ == '__main__':
    config = Config()
    print(config.train_datapath)
    print(config.process_train_datapath)
